from django.contrib.auth.mixins import LoginRequiredMixin
from django.http import JsonResponse


class LoginRequiredJSONMixin(LoginRequiredMixin):
    # 需要改写LoginRequiredMixin中未登录的时候，返回结果的方法
    def handle_no_permission(self):
        return JsonResponse({'code': 400, 'errmsg': '用户未登录！'}, status=401)
